package com.zhang.io;
 
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.text.ParseException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
 
 
public class FileUtils {
 
    /**
     * 压缩文件
     * @param f 原文件
     * @param baseDir 打包根目录
     * @param zos 压缩文件
     * @param deleteSourceFile 是否需要删除原文件，例如：true=删除原文件、false=不删除原文件
     */
    public static void compress(File f, String baseDir, ZipOutputStream zos, boolean deleteSourceFile) {
        if(!f.exists()){
            System.out.println("待压缩的文件目录或文件"+f.getName()+"不存在");
            return;
        }
        File[] fs = f.listFiles();
        if (fs.length==0){

        }
        BufferedInputStream bis = null;
        //ZipOutputStream zos = null;
        byte[] bufs = new byte[1024*10];
        FileInputStream fis = null;
        try{
            //zos = new ZipOutputStream(new FileOutputStream(zipFile));
            for(int i=0; i<fs.length; i++){
                String fName =  fs[i].getName();
                System.out.println("压缩：" + baseDir+fName);
                if(fs[i].isFile()){
                    ZipEntry zipEntry = new ZipEntry(baseDir+fName);//
                    zos.putNextEntry(zipEntry);
                    //读取待压缩的文件并写进压缩包里
                    fis = new FileInputStream(fs[i]);
                    bis = new BufferedInputStream(fis, 1024*10);
                    int read = 0;
                    while((read=bis.read(bufs, 0, 1024*10))!=-1){
                        zos.write(bufs, 0, read);
                    }
                    //如果需要删除源文件，则需要执行下面2句
                    if (deleteSourceFile) {
                        fis.close();
                        fs[i].delete();
                    }
 
                }
                else if(fs[i].isDirectory()){
                    compress(fs[i], baseDir+fName+"/", zos, false);
                }
            }//end for
        }catch  (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } finally{
            //关闭流
            try {
                if (null!=bis) {
                    bis.close();
                }
                //if(null!=zos)
                //zos.close();
                if (null!=fis) {
                    fis.close();
                }
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
 
    }
 
    /**
     * 打包指定路径的文件夹
     * @param sourceFilePath 源文件加地址，例如：C:\\test 必填
     * @param baseDir 打包的根目录名，例如：download/ 非必填
     * @param deleteSourceFile 是否需要删除原文件，例如：true=删除原文件、false=不删除原文件 必填
     */
    public static void compress(String sourceFilePath, String baseDir, boolean deleteSourceFile) {
        File sourceDir = new File(sourceFilePath);
        File zipFile = new File(sourceFilePath+".zip");
        ZipOutputStream zos = null;
        try {
            zos = new ZipOutputStream(new FileOutputStream(zipFile));
            compress(sourceDir, baseDir, zos, deleteSourceFile);
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } finally{
            if (zos!=null) {
                try {
                    zos.close();
                } catch (IOException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
            }
        }
    }

    public static void main(String[] args) {
        compress("D:\\Desktop\\工作\\测试","",false);
    }
 
 
}